package com.yunc.base.exception;

import com.jcraft.jsch.JSchException;
import com.yunc.base.utils.BaseConstant;
import com.yunc.base.utils.R;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.MalformedJwtException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.validation.BindException;
import org.springframework.validation.BindingResult;
import org.springframework.validation.FieldError;
import org.springframework.web.HttpRequestMethodNotSupportedException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
import java.util.Set;

/**
 * 自定义异常处理器
 * 无法捕获MalformedJwtException和ExpiredJwtException，在filter中的异常无法捕获，可以直接通过response返回前端
 *
 * 通过异常直接返回，不需要经过controller是不是有性能问题
 *  if (countUserPostById(postId) > 0){
 *     throw new BusinessException(String.format("%1$s已分配,不能删除", post.getPostName()));
 *  }
 * @author ruoyi
 */
@RestControllerAdvice
public class DefaultExceptionHandler {

    private static final Logger log = LoggerFactory.getLogger(DefaultExceptionHandler.class);
    
    /**
     * token错误，格式
     */
    @ExceptionHandler({ MalformedJwtException.class })
    public R handleMalformedJwtException(MalformedJwtException e) {
        log.error(e.getMessage(), e);
        return R.error(BaseConstant.OAUTH_CODE_ERROR, "token错误，请重新登录");
    }

    /**
     * token异常
     */
    @ExceptionHandler({ ExpiredJwtException.class })
    public R handleExpiredJwtException(ExpiredJwtException e) {
        log.error(e.getMessage(), e);
        return R.error(BaseConstant.OAUTH_CODE_ERROR, "token失效，请重新登录");
    }

    /**
     * 角色权限不够
     */
    @ExceptionHandler({ AccessDeniedException.class })
    public R handleAccessDeniedException(AccessDeniedException e) {
        log.error(e.getMessage(), e);
        return R.error(BaseConstant.OAUTH_CODE_ERROR, "角色权限不够，请联系管理员");
    }

    /**
     * 连接linux异常
     */
    @ExceptionHandler({ JSchException.class })
    public R handleJSchException(JSchException e) {
        log.error(e.getMessage(), e);
        return R.error(BaseConstant.OAUTH_CODE_ERROR, e.getMessage());
    }

    /**
     * 请求方式不支持
     */
    @ExceptionHandler({ HttpRequestMethodNotSupportedException.class })
    public R handleException(HttpRequestMethodNotSupportedException e) {
        log.error(e.getMessage(), e);
        return R.error("不支持' " + e.getMethod() + "'请求");
    }

    /**
     * 拦截未知的运行时异常
     */
    @ExceptionHandler(RuntimeException.class)
    public R notFount(RuntimeException e) {
        log.error("运行时异常:", e);
        return R.error("运行时异常:" + e.getMessage());
    }

    /**
     * 自定义异常
     */
    @ExceptionHandler(UCException.class)
    public R handleUCException(UCException e){
        log.error("运行时异常:", e);
        return R.error(e.getCode(), e.getMsg());
    }

    /**
     * 系统异常
     */
    @ExceptionHandler(Exception.class)
    public R handleException(Exception e) {
        log.error(e.getMessage(), e);
        return R.error("服务器错误，请联系管理员");
    }

    /**
     * 数据校验异常
     */
    @ExceptionHandler(BindException.class)
    public R handleBindException(BindException e) {
        BindingResult bindingResult = e.getBindingResult();
        // getFieldError():随机返回一个对象属性的异常信息。如果要一次性返回所有对象属性异常信息，则调用ex.getAllErrors()
        FieldError fieldError = bindingResult.getFieldError();
        return R.error(fieldError.getDefaultMessage());
    }

    /**
     * 数据校验异常
     */
    @ExceptionHandler(ConstraintViolationException.class)
    public R handleBindException(ConstraintViolationException e) {
        Set<ConstraintViolation<?>> constraintViolations = e.getConstraintViolations();
        StringBuilder sb = new StringBuilder();
        for (ConstraintViolation<?> violation : constraintViolations) {
            sb.append(violation.getMessage());
        }
        return R.error(sb.toString());
    }

    /**
     * 数据校验异常
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public R handleBindException(MethodArgumentNotValidException e) {
        // ex.getFieldError():随机返回一个对象属性的异常信息。如果要一次性返回所有对象属性异常信息，则调用ex.getAllErrors()
        BindingResult bindingResult = e.getBindingResult();
        FieldError fieldError = bindingResult.getFieldError();
        /*StringBuilder sb = new StringBuilder();
        sb.append(fieldError.getField())
                .append("=[")
                .append(fieldError.getRejectedValue())
                .append("]")
                .append(fieldError.getDefaultMessage());*/
        return R.error(fieldError.getDefaultMessage());
    }
}
